import page
from base.base import Base


class PageWebLogin(Base):
    def page_web_username(self, username):
        self.base_input(page.web_login_username, username)

    def page_web_password(self, pwd):
        self.base_input(page.web_login_password, pwd)

    def page_web_verify_code(self, code):
        self.base_input(page.web_login_vertify, code)

    def page_web_login_btn(self):
        self.base_click(page.web_login_submit)

    def page_web_nickname(self):
        return self.base_get_text(page.web_login_nickname)

    def page_web_login(self, username='admin', pwd='123456', code='8888'):
        self.page_web_username(username)
        self.page_web_password(pwd)
        self.page_web_verify_code(code)
        self.page_web_login_btn()
        self.page_web_nickname()
